When Do We Need On-Policy Distillation? Distilling on Offline Student Rollouts Is Often Better
何时需要在线蒸馏?离线学生轨迹蒸馏常更优
🧭 LLM 蒸馏策略 · 在线 vs 离线 rollout
Semi-OPD 用初始学生离线 rollout 蒸馏,在 17 对模型上 14 对优于 OPD,最高 +13.6% 精度、11.4 倍加速
📌 工作参考价值(4 点)
- 做 LLM teacher→student 蒸馏(如 query 改写 / 商品文案 / Agent 策略小模型)时,先别默认 OPD。用一批 hold-out prompt 统计初始师生输出的 top-64 token overlap;若 overlap 不高(论文中多数 < 90%),直接用初始学生离线 rollout 的 Semi-OPD,精度更稳且省 GPU。
- 离线 rollout 可以预生成、缓存、复用,训练时只用固定数据,天然适合电商/广告的离线训练流水线;Semi-OPD 相对 OPD 最高 11.4 倍加速,H100 成本明显下降。
- 若业务必须用 OPD(如需要在线探索),先做 warm start 让学生与老师输出分布接近;否则学生 rollout 随上下文增长会越来越 off-policy,蒸馏信号退化。
- 可以把 token overlap ratio 作为轻量监控指标,在启动蒸馏前判断 OPD 是否值得,避免盲目投入 GPU。